import torch

# 测试电脑的GPU是否可用，True，代表GPU可用，为False代表不可用
print(torch.cuda.is_available())